batch_size=1
soc_version=Ascend310P3
onnx_path=./online_encoder_sim.onnx
output_path=./online_encoder_bs${batch_size}

atc --input_format=ND --framework=5 --model=${onnx_path} \
--input_shape="chunk_xs:${batch_size},67,80;chunk_lens:${batch_size};offset:${batch_size},1;att_cache:${batch_size},12,8,160,128;cnn_cache:${batch_size},12,512,14;cache_mask:${batch_size},1,160" \
--output=${output_path} --log=error --soc_version=${soc_version}
